import pymongo


client = pymongo.MongoClient(host='127.0.0.1', port=27017)
db = client.douyin
task_id = db.task_id

#把文本文件的share_id放进数据库
def handle_init_task():
    with open('douyin1.txt', 'r')as f:
        for i in f.readlines():
            ins = {}
            if i == '\n':
                continue
            ins['share_id'] = i.replace('\n', '')
            task_id.insert(ins)


# handle_init_task()

#把数据库中的share_id拿出来
def handle_get_task():
    return db.task_id.find_one_and_delete({})



#把粉丝的share_id放进数据库
def save_task(task):
    task_id.update({'share_id':task['share_id']},task,True)  #update函数，如果有相同的则更新，没有则插入字典

def save_user_info(user_info):
    db.user.update({'id':user_info['id']},user_info,True)